CM3D2 Converter.misc_VIEW3D_MT_pose_apply

  1# 「3Dビュー」エリア → ポーズモード → Ctrl+A (ポーズ → 適用)
  2import bpy
  3import mathutils
  4from . import common
  5from . import compat
  6
  7
  8# メニュー等に項目追加
  9def menu_func(self, context):
 10    self.layout.separator()
 11    self.layout.operator('pose.apply_prime_field', icon_value=common.kiss_icon())
 12    self.layout.operator('pose.copy_prime_field' , icon_value=common.kiss_icon())
 13
 14@compat.BlRegister()
 15class CNV_OT_copy_prime_field(bpy.types.Operator):
 16    bl_idname = 'pose.copy_prime_field'
 17    bl_label = "Copy Prime Field"
 18    bl_description = "Copies the visual pose of the selected object to the prime field of the active object"
 19    bl_options = {'REGISTER', 'UNDO'}
 20
 21    #is_apply_armature_modifier = bpy.props.BoolProperty(name="Apply Armature Modifier", default=True )
 22    #is_deform_preserve_volume  = bpy.props.BoolProperty(name="Preserve Volume"        , default=True )
 23    #is_keep_original           = bpy.props.BoolProperty(name="Keep Original"          , default=True )
 24    #is_swap_prime_field        = bpy.props.BoolProperty(name="Swap Prime Field"       , default=False)
 25    #is_bake_drivers            = bpy.props.BoolProperty(name="Bake Drivers"           , default=False, description="Enable keyframing of driven properties, locking sliders and twist bones for final apply")
 26    
 27    is_only_selected = bpy.props.BoolProperty(name="Only Selected", default=True )
 28    is_key_location  = bpy.props.BoolProperty(name="Key Location" , default=True )
 29    is_key_rotation  = bpy.props.BoolProperty(name="Key Rotation" , default=True )
 30    is_key_scale     = bpy.props.BoolProperty(name="Key Scale"    , default=True )
 31    is_apply_prime   = bpy.props.BoolProperty(name="Apply Prime"  , default=False, options={'HIDDEN'})
 32    
 33
 34
 35    @classmethod
 36    def poll(cls, context):
 37        target_ob, source_ob = common.get_target_and_source_ob(context)
 38        if target_ob and source_ob:
 39            return True
 40        else:
 41            return False
 42
 43    def invoke(self, context, event):
 44        return context.window_manager.invoke_props_dialog(self)
 45
 46    def draw(self, context):
 47        self.layout.prop(self, 'is_only_selected')
 48        self.layout.prop(self, 'is_key_location' )
 49        self.layout.prop(self, 'is_key_rotation' )
 50        self.layout.prop(self, 'is_key_scale'    )
 51
 52    def execute(self, context):
 53        target_ob, source_ob = common.get_target_and_source_ob(context)
 54        pose = target_ob.pose
 55        arm = target_ob.data
 56
 57        pre_selected_pose_bones = context.selected_pose_bones
 58        pre_mode = target_ob.mode
 59        
 60        bpy.ops.object.mode_set(mode='POSE')
 61        bpy.ops.pose.select_all(action='SELECT')
 62        bpy.ops.pose.constraints_clear()
 63
 64        if not target_ob.pose_library:
 65             bpy.ops.poselib.new()
 66             poselib = target_ob.pose_library
 67
 68        consts = []
 69        bones = pre_selected_pose_bones if self.is_only_selected else pose.bones
 70        for bone in bones:
 71            source_bone = source_ob.pose.bones.get(bone.name)
 72            if source_bone:
 73                if self.is_key_location or self.is_key_rotation:
 74                    const = bone.constraints.new('COPY_TRANSFORMS')
 75                    const.target = source_ob
 76                    const.subtarget = source_bone.name
 77                    consts.append(const)
 78                if self.is_key_scale:
 79                    const = bone.constraints.new('LIMIT_SCALE')
 80                    const.owner_space = 'LOCAL'
 81                    const.use_transform_limit = True
 82                    const.use_min_x = True
 83                    const.use_min_y = True
 84                    const.use_min_z = True
 85                    const.use_max_x = True
 86                    const.use_max_y = True
 87                    const.use_max_z = True
 88                    const.min_x = source_bone.scale.x
 89                    const.min_y = source_bone.scale.y
 90                    const.min_z = source_bone.scale.z
 91                    if source_ob.data.get("is T Stance"):
 92                        source_prime_scale = mathutils.Vector(source_bone.get('prime_scale',(1,1,1)))
 93                        const.min_x *= source_prime_scale.x
 94                        const.min_y *= source_prime_scale.y
 95                        const.min_z *= source_prime_scale.x
 96                    if arm.get("is T Stance"):
 97                        target_prime_scale = mathutils.Vector(bone.get('prime_scale', (1,1,1)))
 98                        const.min_x /= target_prime_scale.x
 99                        const.min_y /= target_prime_scale.y
100                        const.min_z /= target_prime_scale.z
101                    const.max_x = const.min_x
102                    const.max_y = const.min_y
103                    const.max_z = const.min_z
104                    consts.append(const)
105
106        #if True:
107        #    return {'CANCELLED'}
108
109        for i in range(2):
110            is_prime_frame = not bool(i % 2) if arm.get("is T Stance") else bool(i % 2)
111            pose_name = '__prime_field_pose' if is_prime_frame else '__base_field_pose'
112            if self.is_apply_prime:
113                is_prime_frame = not is_prime_frame
114            
115            #if self.is_key_scale and is_prime_frame:
116            #    for const in consts:
117            #        if const.type == 'LIMIT_SCALE':
118            #            const.mute = not is_prime_frame
119            #    bpy.ops.pose.visual_transform_apply()
120            #    for bone in pose.bones:
121            #        bone.keyframe_insert(data_path='scale', frame=i, group=bone.name)
122            #    for const in consts:
123            #        if const.type == 'LIMIT_SCALE':
124            #            const.mute = is_prime_frame
125            
126            for const in consts:
127                const.mute = not is_prime_frame
128            if is_prime_frame:
129                bpy.ops.pose.visual_transform_apply()
130            else:
131                bpy.ops.pose.transforms_clear()
132            for bone in pose.bones:
133                if self.is_key_location:
134                    bone.keyframe_insert(data_path='location'           , frame=i, group=bone.name)
135                if self.is_key_rotation:
136                    bone.keyframe_insert(data_path='rotation_euler'     , frame=i, group=bone.name)
137                    bone.keyframe_insert(data_path='rotation_quaternion', frame=i, group=bone.name)
138                if self.is_key_scale: # and not is_prime_frame:
139                    bone.keyframe_insert(data_path='scale'              , frame=i, group=bone.name)
140                bpy.ops.poselib.pose_add(frame=i, name=pose_name)
141
142        bpy.ops.pose.constraints_clear()
143        bpy.ops.pose.transforms_clear()
144        target_ob.animation_data_clear()
145
146        bpy.ops.pose.select_all(action='DESELECT')
147        if pre_selected_pose_bones:
148            for bone in pre_selected_pose_bones:
149                arm.bones[bone.name].select = True
150
151        if pre_mode: 
152            bpy.ops.object.mode_set(mode=pre_mode)
153        
154        return {'FINISHED'}
155
156
157@compat.BlRegister()
158class CNV_OT_apply_prime_field(bpy.types.Operator):
159    bl_idname = 'pose.apply_prime_field'
160    bl_label = "現在のポーズで素体化"
161    bl_description = "現在のポーズで衣装をモデリングしやすくする素体を作成します"
162    bl_options = {'REGISTER', 'UNDO'}
163
164    is_apply_armature_modifier   = bpy.props.BoolProperty(name="関係するメッシュのアーマチュアを適用", default=True)
165    is_preserve_shape_key_values = bpy.props.BoolProperty(name="Preserve Shape Key Values", default=True , description="Ensure shape key values of child mesh objects are not changed")
166    is_deform_preserve_volume    = bpy.props.BoolProperty(name="アーマチュア適用は体積を維持", default=True)
167    is_keep_original             = bpy.props.BoolProperty(name="Keep Original"            , default=True , description="If the armature is already primed, don't replace the base pose with the current rest pose")
168    is_swap_prime_field          = bpy.props.BoolProperty(name="Swap Prime Field"         , default=False)
169    #is_bake_drivers              = bpy.props.BoolProperty(name="Bake Drivers"             , default=False, description="Enable keyframing of driven properties, locking sliders and twist bones for final apply")
170    
171    
172    was_t_stance = False
173
174    @classmethod
175    def poll(cls, context):
176        ob = context.active_object
177        if ob and ob.type == 'ARMATURE':
178            return True
179        return False
180
181    def invoke(self, context, event):
182        self.was_t_stance = context.object.data.get('is T Stance')
183        return context.window_manager.invoke_props_dialog(self)
184
185    def draw(self, context):
186        self.layout.prop(self, 'is_apply_armature_modifier')
187
188        col = self.layout.column()
189        col.enabled = self.is_apply_armature_modifier
190        col.prop(self , 'is_preserve_shape_key_values')
191        col.prop(self , 'is_deform_preserve_volume'   )
192        if compat.IS_LEGACY:
193            col.prop(prefs, 'custom_normal_blend', icon=compat.icon('SNAP_NORMAL'  ), slider=True)
194
195        self.layout.prop(self, 'is_bake_drivers')
196        if self.was_t_stance:
197            self.layout.prop(self, 'is_keep_original')
198
199    def execute(self, context):
200        ob = context.active_object
201        arm = ob.data
202        pose = ob.pose
203        progress = 0
204
205        pre_selected_objects = context.selected_objects
206        pre_selected_pose_bones = context.selected_pose_bones
207        pre_mode = ob.mode
208        pre_frame = context.scene.frame_current
209
210        bpy.ops.object.mode_set(mode='OBJECT')
211        bpy.ops.object.select_all(action='DESELECT')
212        compat.set_select(ob, True)
213
214        if self.is_swap_prime_field:
215            #context.scene.frame_set(1)
216            bpy.ops.poselib.apply_pose(pose_index=1)
217            bpy.context.view_layer.update()
218
219        if self.is_apply_armature_modifier and ob.children:
220            override = context.copy()
221            context.window_manager.progress_begin(0, len(ob.children)+1)  
222            for child in ob.children:
223                override['object'], override['active_object'] = child, child
224                if child.type == 'MESH' and len(child.modifiers) and bpy.ops.object.forced_modifier_apply.poll(override):
225                    for mod in child.modifiers:
226                        if mod.type == 'ARMATURE':
227                            mod.use_deform_preserve_volume = self.is_deform_preserve_volume
228                            if not mod.object == ob:
229                                had_armature = False
230                            else:
231                                had_armature = True
232                                old_name                = mod.name                      
233                                old_show_expanded       = mod.show_expanded             
234                                old_show_in_editmode    = mod.show_in_editmode          
235                                old_show_on_cage        = mod.show_on_cage              
236                                old_show_render         = mod.show_render               
237                                old_show_viewport       = mod.show_viewport             
238                                old_use_apply_on_spline = mod.use_apply_on_spline       
239                                old_invert_vertex_group = mod.invert_vertex_group       
240                                old_use_bone_envelopes  = mod.use_bone_envelopes        
241                                #old_use_multi_modifier  = mod.use_multi_modifier        
242                                old_use_vertex_groups   = mod.use_vertex_groups         
243                                old_vertex_group        = mod.vertex_group        
244                    apply_results = bpy.ops.object.forced_modifier_apply(override, apply_viewport_visible=True, is_preserve_shape_key_values=self.is_preserve_shape_key_values, initial_progress=progress)
245                    if ('FINISHED' in apply_results) and had_armature:
246                        new_mod = child.modifiers.new(name=old_name, type='ARMATURE')
247                        new_mod.object              = ob
248                        new_mod.use_deform_preserve_volume = self.is_deform_preserve_volume
249                        new_mod.show_expanded       = old_show_expanded      
250                        new_mod.show_in_editmode    = old_show_in_editmode   
251                        new_mod.show_on_cage        = old_show_on_cage       
252                        new_mod.show_render         = old_show_render        
253                        new_mod.show_viewport       = old_show_viewport      
254                        new_mod.use_apply_on_spline = old_use_apply_on_spline
255                        new_mod.invert_vertex_group = old_invert_vertex_group
256                        new_mod.use_bone_envelopes  = old_use_bone_envelopes 
257                        #new_mod.use_multi_modifier  = old_use_multi_modifier 
258                        new_mod.use_vertex_groups   = old_use_vertex_groups  
259                        new_mod.vertex_group        = old_vertex_group
260                
261                progress += 1
262                context.window_manager.progress_update(progress)
263
264        else:
265            context.window_manager.progress_begin(0, 1)  
266
267        temp_ob = ob.copy()
268        temp_arm = arm.copy()
269        temp_ob.data = temp_arm
270        compat.link(context.scene, temp_ob)
271        
272        compat.set_active(context, ob)
273        bpy.ops.object.mode_set(mode='POSE')
274        bpy.ops.pose.select_all(action='SELECT')
275        for bone in ob.pose.bones:
276            prime_scale = mathutils.Vector(bone.get('prime_scale', (1.0,1.0,1.0)))
277            bone_scale = bone.scale #bone.matrix.to_scale()
278            prime_scale.x *= bone_scale.x
279            prime_scale.y *= bone_scale.y
280            prime_scale.z *= bone_scale.z
281            bone['prime_scale'] = prime_scale
282            #bone['_RNA_UI']['prime_scale']['subtype'] = 'XYZ'
283        bpy.ops.pose.armature_apply()
284        bpy.ops.pose.constraints_clear()
285        ob.animation_data_clear()
286        
287        if arm.get("is T Stance") and self.is_keep_original and not self.is_swap_prime_field:
288            anim_data = temp_ob.animation_data
289            if anim_data and anim_data.drivers:
290                drivers = anim_data.drivers
291                for driver in drivers.values():
292                    drivers.remove(driver)
293            #context.scene.frame_set(1)
294            bpy.ops.pose.user_transforms_clear()
295            bpy.ops.poselib.apply_pose(pose_index=1)
296        else:
297            compat.set_active(context, temp_ob)
298            bpy.ops.object.mode_set(mode='POSE')
299            bpy.ops.pose.select_all(action='SELECT')
300            temp_ob.animation_data_clear()
301            bpy.ops.pose.transforms_clear()
302            bpy.ops.object.mode_set(mode='OBJECT')
303            compat.set_select(temp_ob, False)
304        
305        if self.is_swap_prime_field and arm.get('is T Stance'):
306            arm['is T Stance'] = False
307        else:
308            arm['is T Stance'] = True
309
310        # CNV_OT_copy_prime_field.execute()
311        compat.set_select(temp_ob, True)
312        compat.set_active(context, ob)
313        bpy.context.view_layer.update()
314        response = bpy.ops.pose.copy_prime_field(is_only_selected=False, is_key_location=True, is_key_scale=True, is_apply_prime=(not self.is_swap_prime_field))#is_key_location=self.is_bake_drivers, is_key_scale=self.is_bake_drivers, is_apply_prime=True)
315
316        context.window_manager.progress_end()
317
318        if not 'FINISHED' in response:
319            return response
320
321        common.remove_data(temp_arm)
322        try:
323            common.remove_data(temp_ob)
324        except:
325            pass
326        
327        bpy.ops.object.mode_set(mode='POSE')
328        bpy.ops.pose.select_all(action='DESELECT')
329        if pre_selected_pose_bones:
330            for bone in pre_selected_pose_bones:
331                arm.bones[bone.name].select = True
332
333        if pre_selected_objects:
334            for o in pre_selected_objects:
335                compat.set_select(o, True)
336        compat.set_active(context, ob)
337        bpy.ops.object.mode_set(mode=pre_mode)
338
339        context.scene.frame_set(pre_frame)
340
341
342
343        return {'FINISHED'}